home *** CD-ROM | disk | FTP | other *** search
/ Clickx 115 / Clickx 115.iso / software / tools / windows / tails-i386-0.16.iso / live / filesystem.squashfs / usr / share / pyshared / whisperBack / utils.py < prev    next >
Encoding:
Python Source  |  2012-12-13  |  4.9 KB  |  170 lines
  1. #!/usr/bin/env python
  2. # -*- coding: UTF-8 -*-
  3.  
  4. ########################################################################
  5. # WhisperBack - Send feedback in an encrypted mail
  6. # Copyright (C) 2009-2012 Tails developers <amnesia.org>
  7. #
  8. # This file is part of WhisperBack
  9. #
  10. # WhisperBack is  free software; you can redistribute  it and/or modify
  11. # it under the  terms of the GNU General Public  License as published by
  12. # the Free Software Foundation; either  version 3 of the License, or (at
  13. # your option) any later version.
  14. # 
  15. # This program  is distributed in the  hope that it will  be useful, but
  16. # WITHOUT   ANY  WARRANTY;   without  even   the  implied   warranty  of
  17. # MERCHANTABILITY  or FITNESS  FOR A  PARTICULAR PURPOSE.   See  the GNU
  18. # General Public License for more details.
  19. #
  20. # You should have received a copy of the GNU General Public License
  21. # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22. ########################################################################
  23.  
  24. """various WhisperBack utility functions
  25.  
  26. """
  27.  
  28. import os
  29. import re
  30. import urlparse
  31. import locale
  32.  
  33. # Ugly pathes finder utilities
  34.  
  35. def guess_prefix ():
  36.     """Tries to guess the prefix
  37.  
  38.     @return The guessed prefix"""
  39.  
  40.     # XXX: hardcoded path !
  41.     if os.path.exists ("/usr/local/share/whisperback"):
  42.         return "/usr/local"
  43.     elif os.path.exists ("/usr/share/whisperback"):
  44.         return "/usr"
  45.     else:
  46.         return None
  47.  
  48. def get_sharedir ():
  49.     """Tries to guess the shared data directiry
  50.  
  51.     @return The guessed shared data directiry"""
  52.  
  53.     if guess_prefix():
  54.         return os.path.join (guess_prefix(), "share")
  55.     else:
  56.         return "data"
  57.  
  58. def get_datadir ():
  59.     """Tries to guess the datadir
  60.  
  61.     @return The guessed datadir"""
  62.     if guess_prefix():
  63.         return os.path.join (get_sharedir(), "whisperback")
  64.     else:
  65.         return "data"
  66.  
  67. def get_pixmapdir ():
  68.     """Tries to guess the pixmapdir
  69.  
  70.     @return The guessed pixmapdir"""
  71.  
  72.     if guess_prefix():
  73.         return os.path.join (get_sharedir(), "pixmaps")
  74.     else:
  75.         return "data"
  76.  
  77. # Input validation fuctions
  78.  
  79. def is_valid_link(candidate):
  80.     """Check if candidate seems to be a internet link
  81.  
  82.     @param candidate the URL to be checked
  83.  
  84.     @returns true if candidate is an URL with:
  85.     - an hostname of the form domain.tld
  86.     - a scheme http(s) or ftp(S)
  87.     """
  88.     parseresult = urlparse.urlparse(candidate)
  89.     #pylint: disable=E1101
  90.     if (re.search(r'^(ht|f)tp(s)?$', parseresult.scheme) and
  91.         re.search(r'^(\w{1,}\.){1,}\w{1,}$', parseresult.hostname)):
  92.         return True
  93.     else:
  94.         return False
  95.  
  96. def is_valid_pgp_block(candidate):
  97.     """Check if candidate seems to be a PGP public key block
  98.  
  99.     @param    candidate the string to be checked
  100.  
  101.     @returns  true if candidate starts with `-----BEGIN PGP PUBLIC KEY BLOCK----`
  102.               and ends with `-----END PGP PUBLIC KEY BLOCK-----`
  103.     """
  104.     #pylint: disable=C0301
  105.     if re.search(r"-----BEGIN PGP PUBLIC KEY BLOCK-----\n(?:.*\n)+-----END PGP PUBLIC KEY BLOCK-----",
  106.             candidate):
  107.         return True
  108.     else:
  109.         return False
  110.  
  111. def is_valid_pgp_id(candidate):
  112.     """Check if candidate looks like a pgp key ID
  113.  
  114.     @param    candidate the string to be checked
  115.  
  116.     @returns  true if candidate is either an 8 or 16 digit hex number or a 40
  117.               digit hex fingerprint
  118.     """
  119.     #pylint: disable=C0301
  120.     if re.search(r"(?:^(?:0x)?(?:[0-9a-fA-F]{8}){1,2}$)|(?:^(?:[0-9f-zA-F]{4} {0,2}){10}$)",
  121.             candidate):
  122.         return True
  123.     else:
  124.         return False
  125.  
  126. def is_valid_email(candidate):
  127.     """Check if candidate looks like an email address
  128.  
  129.     @param    candidate the string to be checked
  130.  
  131.     @returns  true if candidate is in the form test@example.com
  132.     """
  133.     if re.search(r"[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Za-z]{2,4}", candidate):
  134.         return True
  135.     else:
  136.         return False
  137.  
  138. def sanitize_hardware_info(log_string):
  139.     """Sanitize hardware-identifying info from a string
  140.  
  141.     Removes strings:
  142.     
  143.     - labeled as serial numbers and UUID;
  144.     - looking like IPs or MAC addresses.
  145.  
  146.     @param  log_string  the string to be sanitized
  147.  
  148.     @returns a sanitized version of log_string
  149.     """
  150.     # XXX: must be updated once IPv6 is enabled
  151.  
  152.     # Serial Numbers
  153.     log_string = re.sub(r'((Serial Number:?[\s]+|SerialNo=|iSerial[\s]+[\d]+\s+|SerialNumber:|Serial#:)[\s]+)[^\s].+',
  154.                         r'\1[SN REMOVED]',
  155.                         log_string)
  156.     # UUIDs
  157.     log_string = re.sub(r'(UUID:[\s]+)[^\s].+',
  158.                         r'\1[UUID REMOVED]',
  159.                         log_string)
  160.  
  161.     # IPs
  162.     log_string = re.sub(r'\<([\d]{1,3}\.){3}[\d]{1,3}\>',
  163.                         r'[IP REMOVED]',
  164.                         log_string)
  165.     # MAC addresses
  166.     log_string = re.sub(r'\<([0-9a-fA-F]{2}:){5,}[0-9a-fA-F]{2}\>',
  167.                         r'[MAC REMOVED]',
  168.                         log_string)
  169.     return log_string
  170.